adapt_alg="TAST-BN"
adapt_alg_array=("ERM" "BN" "Tent" "PL" "SHOT-IM" "T3A" "TAST" "TAST-BN" "TSD" "PROGRAM" "DeYO")
#adapt_alg_array=("TAST-BN")
dataset="cifar10"
corruption="shot_noise"
#corruption_array=("shot_noise" "motion_blur" "snow" "pixelate" "gaussian_noise" "defocus_blur" "brightness" "fog" "zoom_blur" "frost" "glass_blur" "impulse_noise" "contrast" "jpeg_compression" "elastic_transform")
corruption_array=("shot_noise")
test_envs=0
network="resnet18"
lr=1e-4
seed=0
seed_array=(0)
gpu_id=0

for corruption in ${corruption_array[@]}
do
    for adapt_alg in ${adapt_alg_array[@]}
    do
        echo "-------------------"
        echo "$corruption $adapt_alg"
        CUDA_VISIBLE_DEVICES=${gpu_id} python unsupervise_adapt_cifar10c.py --corruption ${corruption} --dataset ${dataset} --data_dir /data/huyihao/projects/data/${dataset} --adapt_alg ${adapt_alg} --pretrain_dir train_output/${network}/${dataset}/${domain}/${seed}/model.pkl --lr ${lr} --net ${network} --test_envs ${test_envs} --seed ${seed} --output tta_output/${network}/${dataset}/${domain}/${adapt_alg}
    done
    python print.py
done